from .partA2_head import PartA2FCHeadTS